package com.ai.dmc.utils.export.excelhelper;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

import org.apache.log4j.Logger;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

public class TExcelFactory<T> {
	
	private short DEFAULTCOLUMNWIDTH = 15;
	
	private String DEFAULTDATEPATTERN = "yyyy-MM-dd";
	
	private static final Logger logger = Logger.getLogger(TExcelFactory.class);
	
	/**
	 *  这是一个通用的方法，利用了JAVA的反射机制，可以将放置在JAVA集合中并且符号一定条件的数据以EXCEL形式输出
	 *  创建单行表头的Excel
	 * @param sheetName：表单标签名称
	 * @param headers：表单表头，单行表头
	 * @param dataset：数据集，以类对象的方式组织
	 * @param methodNames：获取类对象属性的方法
	 * @param pattern：如果有时间数据，设定输出格式，可以为null，当为null时，默认采用“yyyy-MM-dd“
	 * @return
	 */
	public HSSFWorkbook createWorkbookByObj(String sheetName, String[] headers,
			Collection<T> dataset, String[] methodNames, String pattern) {
		// 声明一个工作薄
		HSSFWorkbook workbook = new HSSFWorkbook();
		// 生成一个表格
		HSSFSheet sheet = workbook.createSheet(sheetName);
		// 设置表格默认列宽度
		sheet.setDefaultColumnWidth(DEFAULTCOLUMNWIDTH);
		//设置表头
		POIExcelHelper.parseTitle(sheet, POIExcelHelper.getDefaultStyle(workbook),headers);
		this.parseDataByObj(sheet, dataset, methodNames, POIExcelHelper.getDefaultFootStyle(workbook), pattern);
		return workbook;
	}

	/**
	 * 创建多行表头的Excel
	 * @param sheetName：表单的标签
	 * @param headerList：表单表头，必须遵循CellBean的要求
	 * @param dataset：数据集，以类对象的方式组织
	 * @param methodNames：获取类对象属性的方法
	 * @param pattern：日期格式，可以为null，当为null时，采用默认格式”yyyy-MM-dd“
	 * @return
	 */
	public HSSFWorkbook createWorkbookByObj(String sheetName, List<CellBean> headerList,
			Collection<T> dataset, String[] methodNames, String pattern) {
		// 声明一个工作薄
		HSSFWorkbook workbook = new HSSFWorkbook();
		// 生成一个表格
		HSSFSheet sheet = workbook.createSheet(sheetName);
		// 设置表格默认列宽度
		sheet.setDefaultColumnWidth(DEFAULTCOLUMNWIDTH);
		//设置表头
		POIExcelHelper.parseTitle(sheet, POIExcelHelper.getDefaultStyle(workbook),headerList);
		int headerRowSize = this.getHeadRowSize(headerList);
		this.parseDataByObj(sheet, dataset, methodNames, POIExcelHelper.getDefaultFootStyle(workbook), pattern,headerRowSize);
		return workbook;
	}	
	/**
	 * 创建Excel表格，数据是直接以Object的方式组织后，存在Collection中的
	 * @param sheet
	 * @param dataset：
	 * @param methodNames：获取Object对象中属性的方法，可以为null，当为null时，将根据类反射机制，从bean中自动查找方法
	 *                     注意：如果不指定methodNames，那么顺序将按照属性在类中的顺序获取，所以表头的顺序一定要与此顺序相同
	 * @param style：表格的样式
	 * @param pattern：时间格式，默认为“yyyy-MM-dd”
	 */
	public void parseDataByObj(HSSFSheet sheet, Collection<T> dataset, String[] methodNames, HSSFCellStyle style, String pattern){
		parseDataByObj(sheet,dataset,methodNames,style,pattern,1);
	}
	
	/**
	 * 创建Excel表格，数据是直接以Object的方式组织后，存在Collection中的
	 * @param sheet
	 * @param dataset：
	 * @param methodNames：获取Object对象中属性的方法，可以为null，当为null时，将根据类反射机制，从bean中自动查找方法
	 *                     注意：如果不指定methodNames，那么顺序将按照属性在类中的顺序获取，所以表头的顺序一定要与此顺序相同
	 * @param style：表格的样式
	 * @param pattern：时间格式，默认为“yyyy-MM-dd”
	 */
	public void parseDataByObj(HSSFSheet sheet, Collection<T> dataset, String[] methodNames, HSSFCellStyle style, String pattern,int headerRosSize){
		// 遍历集合数据，产生数据行
		Iterator<T> it = dataset.iterator();
		int index = headerRosSize;
		while (it.hasNext()) {
			HSSFRow row = sheet.createRow(index);
			T t = (T) it.next();
			if(methodNames == null){
				methodNames = this.getMethodNames(t);
			}
			for(short i=0; i<methodNames.length; i++){
				try {
					Class tCls = t.getClass();
					Method getMethod = tCls.getMethod(methodNames[i],new Class[] {});
					Object value = getMethod.invoke(t, new Object[] {});
					// 判断值的类型后进行强制类型转换
					String textValue = null;
					if (value instanceof Boolean) {
						boolean bValue = (Boolean) value;
						textValue = "1";
						if (!bValue) {
							textValue = "0";
						}
					} else if (value instanceof Date) {
						Date date = (Date) value;
						if(pattern == null){
							pattern = DEFAULTDATEPATTERN;
						}
						SimpleDateFormat sdf = new SimpleDateFormat(pattern);
						textValue = sdf.format(date);
					}else {
						// 其它数据类型都当作字符串简单处理
						if(value==null){//huangc modify 20100417 增加对null值的处理
							textValue="";
						}else{
							textValue = value.toString();
						}
					}
					// 如果不是图片数据，就利用正则表达式判断textValue是否全部由数字组成
					if (textValue != null) {
						POIExcelHelper.createCell(row, i, textValue, style);
					}
				} catch (SecurityException se) {
					logger.error("reflect failed!");
					logger.debug("TExcelFactory.parseDataByObj():",se);
				} catch (NoSuchMethodException nme) {
					logger.error("reflect failed, maybe method not found!");
					logger.debug("TExcelFactory.parseDataByObj():",nme);
				} catch (IllegalArgumentException iae) {
					logger.error("reflect failed, maybe parameter invalid!");
					logger.debug("TExcelFactory.parseDataByObj():",iae);
				} catch (IllegalAccessException iae) {
					logger.error("reflect failed, access denied!");
					logger.debug("TExcelFactory.parseDataByObj():",iae);
				} catch (InvocationTargetException ite) {
					logger.error("reflect failed, return object invalid!");
					logger.debug("TExcelFactory.parseDataByObj():",ite);
				} 
			}
			index++;
		}
	}
	
	/**
	 * 获取泛型类（Bean）中所有的取得属性值的方法
	 * 其中获取的方法遵循的原则是：boolean性的以“is”开头，其它的均以“get”开头
	 * @param t
	 * @return
	 */
	private String[] getMethodNames(T t){
		// 利用反射，根据JavaBean属性的先后顺序，动态调用getXxx()方法得到属性值
		Field[] fields = t.getClass().getDeclaredFields();
		String[] methodNames = new String[fields.length];
		for (int i = 0; i < fields.length; i++) {
			Field field = fields[i];
			String fieldName = field.getName();
			String prefix = "get";
			if(field.getType().equals(boolean.class)){
				prefix = "is";
			}
			String getMethodName = prefix
				+ fieldName.substring(0, 1).toUpperCase()
				+ fieldName.substring(1);
			methodNames[i] = getMethodName;
		}
		return methodNames;
	}
	
	/**
	 * 获取表头所占的行数
	 * @param headerList
	 * @return
	 */
	private int getHeadRowSize(List<CellBean> headerList){
		int rowSize = 1;
		for(CellBean cellBean : headerList){
			if((cellBean.getRowindex()+1) > rowSize){
				rowSize = cellBean.getRowindex() + 1;
			}
		}
		return rowSize;
	}
	
//	public static void main(String[] args) throws IOException {
		// 测试
//		String title = "测试表单";
//		TExcelFactory<UserCity> ex = new TExcelFactory<UserCity>();
//		String[] headers = { "城市ID", "城市名称", "部门ID", "父ID"};
//		List<UserCity> dataset = new ArrayList<UserCity>();
//		for(int i=0; i<10; i++){
//			UserCity uc = new UserCity();
//			uc.setCityid(Integer.toString(i));
//			uc.setCityname("城市"+i);
//			uc.setDmDeptId("部门" +i);
//			uc.setParentid("父" + i);
//			dataset.add(uc);
//		}
//		String sheetName = "对象生成Excel表";
//		String[] methodNames = {"getCityid","getCityname","getDmDeptId","getParentid"};
//		HSSFWorkbook workbook = ex.createWorkbookByObj(sheetName, headers, dataset, methodNames, null);
//		String dirPath = System.getProperty("user.home") + "\\TEMPEXCELFILE";
//		String fileName = "对象生成Excel表.xls";
//		POIExcelHelper.save(workbook, dirPath, fileName);
//	}
}
